package com.bcbim.common.util;

import net.sf.jxls.transformer.XLSTransformer;
import org.apache.poi.hssf.usermodel.*;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFCell;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.util.List;
import java.util.Map;


public class ExcelUtil {

    /**
     * @param workbook
     * @param fontSize
     * @return 表头单元格样式
     */
    private static CellStyle createHeadCellStyle(Workbook workbook, short fontSize) {
        //创建cellStyle对象
        CellStyle style = workbook.createCellStyle();
        //设置边框
        style.setBorderBottom(CellStyle.BORDER_THIN);
        style.setBorderLeft(CellStyle.BORDER_THIN);
        style.setBorderRight(CellStyle.BORDER_THIN);
        style.setBorderTop(CellStyle.BORDER_THIN);
        //设置文字居中
        style.setAlignment(CellStyle.ALIGN_CENTER);
        //设置字体大小、加粗
        Font font = workbook.createFont();
        font.setFontHeightInPoints(fontSize);
        //设置文字样式生效
        style.setFont(font);
        return style;
    }

    /**
     * @param workbook
     * @param fontSize
     * @return 表信息单元格样式
     */
    private static CellStyle createMesCellStyle(Workbook workbook, short fontSize) {
        //创建cellStyle对象
        CellStyle style = workbook.createCellStyle();
        //设置边框
        style.setBorderBottom(CellStyle.BORDER_THIN);
        style.setBorderLeft(CellStyle.BORDER_THIN);
        style.setBorderRight(CellStyle.BORDER_THIN);
        style.setBorderTop(CellStyle.BORDER_THIN);
        //设置文字居中
        style.setAlignment(CellStyle.ALIGN_CENTER);
        //设置字体大小以及字体水平&垂直居中
        Font font = workbook.createFont();
        style.setVerticalAlignment(CellStyle.VERTICAL_CENTER);
        font.setFontHeightInPoints(fontSize);
        //设置自动换行
        style.setWrapText(true);
        //设置文字样式生效
        style.setFont(font);
        return style;
    }

    /**
     * 自适应宽度(中文支持)
     *
     * @param sheet
     * @param size
     */
    private static void setSizeColumn(Sheet sheet, int size) {
        for (int columnNum = 0; columnNum < size; columnNum++) {
            int columnWidth = sheet.getColumnWidth(columnNum) / 256;
            for (int rowNum = 0; rowNum < sheet.getLastRowNum(); rowNum++) {
                Row currentRow;
                //当前行未被使用过
                if (sheet.getRow(rowNum) == null) {
                    currentRow = sheet.createRow(rowNum);
                } else {
                    currentRow = sheet.getRow(rowNum);
                }

                if (currentRow.getCell(columnNum) != null) {
                    Cell currentCell = currentRow.getCell(columnNum);
                    if (currentCell.getCellType() == XSSFCell.CELL_TYPE_STRING) {
                        int length = currentCell.getStringCellValue().getBytes().length;
                        if (columnWidth < length) {
                            columnWidth = length;
                        }
                    }
                }
            }
            sheet.setColumnWidth(columnNum, columnWidth * 256);
        }
    }

    /**
     * 导出Excel
     *
     * @param sheetName sheet名称
     * @param title     标题
     * @param values    内容，可以修改为list  或者 map
     * @param wb        HSSFWorkbook对象
     * @return
     */
    public static HSSFWorkbook getHSSFWorkbook(String sheetName, String[] title, String[][] values, HSSFWorkbook wb) {
        // 第一步，创建一个HSSFWorkbook，对应一个Excel文件
        if (wb == null) {
            wb = new HSSFWorkbook();
        }

        // 第二步，在workbook中添加一个sheet,对应Excel文件中的sheet
        HSSFSheet sheet = wb.createSheet(sheetName);
        //设置列宽:
        sheet.autoSizeColumn(0);
        // 第三步，在sheet中添加表头第0行,注意老版本poi对Excel的行数列数有限制
        HSSFRow row = sheet.createRow(0);

        // 第四步，创建单元格，并设置值表头 设置表头居中
        HSSFCellStyle style = wb.createCellStyle();
        style.setAlignment(HSSFCellStyle.ALIGN_CENTER); // 创建一个居中格式

        //声明列对象
        HSSFCell cell = null;

        //创建标题
        for (int i = 0; i < title.length; i++) {
            cell = row.createCell(i);
            cell.setCellValue(title[i]);
            cell.setCellStyle(createHeadCellStyle(wb, (short) 12));
        }
        //创建内容
        for (int i = 0; i < values.length; i++) {
            row = sheet.createRow(i + 1);
            for (int j = 0; j < values[i].length; j++) {
                //将内容按顺序赋给对应的列对象
                cell = row.createCell(j);
                cell.setCellValue(values[i][j]);
                cell.setCellStyle(createHeadCellStyle(wb, (short) 10));
            }
        }
        setSizeColumn(sheet, values.length + 3);
        return wb;
    }

    /**
     * 根据模板导出数据
     *
     * @param templateFile 导出模板
     * @param file         目标文件
     * @param beanParams   导出参数
     * @throws Exception
     */
    public static void downLoadExcel(File templateFile, File file, Map<String, Object> beanParams)
            throws Exception {
        try {
            //方案二：将导出内容存储到本地，返回文件路径
            FileOutputStream os = new FileOutputStream(file);
            //读取模板
            InputStream is = new FileInputStream(templateFile);
            XLSTransformer transformer = new XLSTransformer();
            //向模板中写入内容
            Workbook workbook = transformer.transformXLS(is, beanParams);
            //获取工作表
            Sheet sheet = workbook.getSheet("Sheet1");
            List<?> list = (List<?>) beanParams.get("list");
            setSizeColumn(sheet, list.size() + 3);
            sheet.setColumnWidth(2,4000);
            sheet.setColumnWidth(3,5000);
            sheet.setColumnWidth(8,4000);
            sheet.setColumnWidth(9,4000);
            //写入成功后转化为输出流
            workbook.write(os);
            os.close();
        } catch (Exception e) {
            e.printStackTrace();
            throw e;
        }
    }
}

